TileMix: Tile-Centric Mixed-Precision Attention for LLM Inference Acceleration
TileMix:面向 LLM 推理加速的 Tile 中心混合精度注意力
🧭 LLM 推理 · 混合精度注意力 kernel ★ 深度解读
University of North Texas · Saint Louis University
通过 tile 分组路由 FP16/INT8 注意力计算,在保留 dense 连接的同时加速长上下文 prefill 并恢复质量
📌 工作参考价值(3 点)
- 长上下文 RAG / Agent memory / 多跳检索场景中,prefill 是延迟大头;可借鉴按 attention score tile 分组做混合精度(关键区域 FP16,其余 INT8),而不是整层 uniform INT8,能在质量和吞吐之间拿到更好 trade-off,且无需训练或改模型权重。
- 路由模板可直接复用稀疏注意力的空间先验(local band / global / random / SpTrans / BigBird),但只决定精度路径、不删除 token 交互;业务上可先用静态模板快速验证哪些位置对效果敏感,再考虑升级为自适应或 learnable 路由。
- INT8 KV cache + per-block scale 能降低显存和 decode 成本,同时 V 保持 FP16 做 PV 更新,工程风险较低;如果使用 FlashAttention-style serving,可参考其 shared online-softmax 状态与 bitmask constant-time 查找的集成方式,支持 GQA 和 variable-length batching。